from langchain.agents import tool

import json
from config import web_json


@tool()
def save() -> str:
    """
    自动根据已知数据生成网元，不用传入参数

    Returns:
        str: 保存结果消息
    """
    # 将新数据添加到列表
    # web_json.append(json_data)

    print("json_data", web_json)

    # 将所有数据连接起来写入文件（根据你的需求可能需要调整）
    # with open("web.json", "w", encoding="utf-8") as file:
    #     file.write("\n".join(web_json))  # 假设每条数据用换行符分隔
    with open("./web/public/data.json", "w", encoding="utf-8") as file:
        json.dump(web_json, file, ensure_ascii=False, indent=2)  # 使用json模块直接写入

    return "保存网页成功：http://localhost:1001"
